package 每日一题;

import java.util.Map;
import java.util.TreeMap;

/**
 * @description:
 * @author: 小白白
 * @create: 2021-11-23
 **/

public class No594最长和谐子序列 {

    /**
     * 和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。
     * 现在，给你一个整数数组 nums ，请你在所有可能的子序列中找到最长的和谐子序列的长度。
     * 数组的子序列是一个由数组派生出来的序列，它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。
     * <p>
     * 示例 1：
     * 输入：nums = [1,3,2,2,5,2,3,7]
     * 输出：5
     * 解释：最长的和谐子序列是 [3,2,2,2,3]
     * 示例 2：
     * 输入：nums = [1,2,3,4]
     * 输出：2
     * 示例 3：
     * 输入：nums = [1,1,1,1]
     * 输出：0
     *  
     * 提示：
     * 1 <= nums.length <= 2 * 104
     * -109 <= nums[i] <= 109
     */

    public int findLHS(int[] nums) {

        TreeMap<Integer, Integer> treeMap = new TreeMap<>();

        for (int num : nums) {
            treeMap.put(num, treeMap.getOrDefault(num, 0) + 1);
        }

        Map.Entry<Integer, Integer> entry = treeMap.pollFirstEntry();
        // 上一个数字
        int lastNum = entry.getKey();
        // 上一个数字的个数
        int lastCount = entry.getValue();

        int result = 0;

        for (Map.Entry<Integer, Integer> e : treeMap.entrySet()) {
            int k = e.getKey();
            int v = e.getValue();
            // 连续的数字,可以融合last
            if ((lastNum + 1) == k) {
                // 建议直接融合lastSum
                result = Math.max(result, lastCount + v);
            }
            // 非连续的数字,不可更新,因为必须差值为1呀
            lastNum = k;
            lastCount = v;
        }

        return result;
    }

    public static void main(String[] args) {
        No594最长和谐子序列 n = new No594最长和谐子序列();
        int[] arr = {1, 1, 1, 1};
        int result = n.findLHS(arr);
        System.out.println(result);
    }

}
